New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Fix Dockerfile #34

Merged

yujincheng08 merged 1 commit into NVlabs:main from JimothyJohn:docker

Nov 24, 2024

Contributor

JimothyJohn commented Nov 23, 2024

Added necessary dependencies and fixed formatting warnings:

Dependencies: Added libgl1-mesa-glx and libglib2.0-0
Custom libraries: COPY tools/ folder as well
ENV formatting, add = as recommended
CMD - Convert to JSON array formatting (removed last comma) and fix model path.

Recommend also updating README

# Alternatively, set up with Docker
docker build -t sana:latest .
# Add your HF_TOKEN to allow Gemma download
# Set your HF_HOME to avoid downloading each build
docker run --gpus all \
  --ulimit memlock=-1 \
  --ulimit stack=67108864 \
  -v "$HF_HOME"=/root/.cache/huggingface \
  -e HF_TOKEN=hf_xxxxxxx \
  sana:latest


          Fix Dockerfile

9740a9b

yujincheng08 merged commit ee5b369 into NVlabs:main

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet